#sigmoid函数的实现，sigmoid函数与阶跃函数的比较
import numpy as np
import matplotlib.pyplot as plt

def step_function(x):  #定义阶跃函数
    return np.array(x>0, dtype=int)

def sigmoid_function(x):  #定义sigmoid函数
    return 1 / (1 + np.exp(-x))

x = np.arange(-5.0, 5.0, 0.1)
y1 = sigmoid_function(x)
y2 = step_function(x)
plt.plot(x, y1, linestyle='--')
plt.plot(x, y2)
plt.ylim(-0.1, 1.1)  #指定y轴范围
plt.show()
